import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

names = ['成色', '市场价格', '颜色', '材质', '是否限量', '回收价格']
df = pd.read_csv('data.csv', names=names, sep='\t', header=0)

# 设置X轴标签
plt.xlabel('市场价格')
# 设置Y轴标签
plt.ylabel('回收价格')
# 设置图标题
plt.title('成色-市场价格-回收价格散点图')

alpha = 1

# 99 S级数据
df_99 = df[df.成色 == 99]
# 绘制散点图
plt.scatter(df_99.市场价格, df_99.回收价格, c='b', marker='o', s=10, alpha=alpha)

# 90 A级数据
df_90 = df[df.成色 == 90]
# 绘制散点图
plt.scatter(df_90.市场价格, df_90.回收价格, c='r', marker='o', s=10, alpha=alpha)

# 80 B级数据
df_80 = df[df.成色 == 80]
# 绘制散点图
plt.scatter(df_80.市场价格, df_80.回收价格, c='c', marker='o', s=10, alpha=alpha)

# 80以下 C级数据
df_60 = df[df.成色 == 60]
# 绘制散点图
plt.scatter(df_60.市场价格, df_60.回收价格, c='k', marker='o', s=10, alpha=alpha)

# 设置图标
plt.legend(['99成新',
            '9成新',
            '8成新',
            '8成新以下'])
plt.show()

